from torch.utils.data import DataLoader, Dataset


class CustomDataset(Dataset):
    def __init__(self):
        super(CustomDataset, self).__init__()
